# 引入第三方库
from traceback import format_tb
from openpyxl import load_workbook
import stu
# 打开相应的excle表格   相对路径  绝对路径 
wb_us = load_workbook(filename="C:/Users/G15/Desktop/user.xlsx")
# 定义一个登录的方法
def login():
    # print("成功")
    # 打开子表
    wb_if = wb_us['用户信息']

    # 输入用户名和密码
    user_name = input("请输入用户名：")
    user_pwd = int(input("请输入密码："))

    # 依次遍历相应的用户表格里面的数据
    # wb_if.max_row+1 不包含端点值 所有+1
    for i in range(2,wb_if.max_row+1):
        # 判断用户名
        if user_name == wb_if[i][0].value:
            print("登录成功")
            # 判断密码
            if user_pwd == wb_if[i][1].value:
                print("登录成功")
                stu.stu_menu()
            else:
                print("密码错误")

# 定义一个注册的方法
def again():
    # print("注册成功")
    # 用户名和密码
    wb_if =wb_us['用户信息']
    # 用户名和密码
    user_name_re = input("请输入新注册的用户名：")
    user_pwd_re = int(input("请输入新注册的密码："))
    user_pwd = input("请输入刚注册的真实密码：")

    # 判断
    if user_pwd == user_pwd_re:
        user_name_zx = input("请输入新注册的真实姓名：")
        user_name_zh = input("请输入证件号：")
        # 定义为列表
        info = [user_name_re,user_pwd,user_name_zx,user_name_zh]
        # 追加到相应的表格
        wb_if.append(info)
        wb_us.save(filename="C:/Users/G15/Desktop/user.xlsx")
        print("恭喜你注册成功")

# 定义一个忘记密码的方法
def forget():
    # print("忘记密码成功")
    wb_if =wb_us['用户信息']
    user_name = input('请输入用户名：')
    while True:
        flag = False
        for i in range(2,wb_if.max_row+1):
            if wb_if[i][0].value == user_name:
                flag = True
                true_name = input('请输入该账户的真实姓名：')
                user_id = input('请输入该账户的证件号：')
                if wb_if[i][2].value == true_name and wb_if[i][3].value == user_id:
                    user_pwd = input('请输入新密码：')
                    user_pwd_re = input('请输入新密码：')
                    if user_pwd == user_pwd_re:
                        wb_if[i][1].value = user_pwd
                        wb_us.save(filename="C:/Users/G15/Desktop/user.xlsx")
                        print('密码修改成功')
                        return
                    else:
                        print('两次输入的密码不一致！')
                else:
                    print('用户信息错误！')
        if not flag:
            print('查无该号！')
            user_name = input("请重新输入用户名：")

